package tests.mocks;

import java.util.ArrayList;
import java.util.Collection;

import vuelo.VueloConEscala;
import vuelo.VueloDirecto;

public class VueloConEscalaMock extends VueloConEscala {
	
	
	public VueloConEscalaMock (VueloConEscala vueloConEscala){
		super(vueloConEscala.getVuelosDirectos());
		this.mockearVuelosDirectos();
	}


	public VueloConEscalaMock(VueloDirecto vueloDirectoOrigen, VueloDirecto vueloDirectoDestino) {
		super(vueloDirectoOrigen, vueloDirectoDestino);
		this.mockearVuelosDirectos();
	}
	
	private void mockearVuelosDirectos() {
		Collection<VueloDirectoMock> vuelosMockeados = new ArrayList<VueloDirectoMock>();
		for (VueloDirecto vueloDirecto: this.getVuelosDirectos()) {
			vuelosMockeados.add(new VueloDirectoMock((VueloDirecto)vueloDirecto));
		}
		this.setVuelosDirectos(new ArrayList<VueloDirecto>());
		this.getVuelosDirectos().addAll(vuelosMockeados);
	}

	
	





}
